# versions.py -- core Portage functionality
# Copyright 1998-2023 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2

__all__ = [
    "best",
    "catpkgsplit",
    "catsplit",
    "cpv_getkey",
    "cpv_getversion",
    "cpv_sort_key",
    "pkgcmp",
    "pkgsplit",
    "ververify",
    "vercmp",
]

import re
import typing
import warnings
from functools import lru_cache
from typing import Any, Optional, Union
from collections.abc import Sequence


import portage

portage.proxy.lazyimport.lazyimport(
    globals(),
    "portage.repository.config:_gen_valid_repo",
    "portage.util:cmp_sort_key",
)
from portage import _unicode_decode
from portage.eapi import _eapi_attrs, _get_eapi_attrs
from portage.exception import InvalidData
from portage.localization import _

_unknown_repo = "__unknown__"

# \w is [a-zA-Z0-9_]

# PMS 3.1.3: A slot name may contain any of the characters [A-Za-z0-9+_.-].
# It must not begin with a hyphen or a dot.
_slot = r"([\w+][\w+.-]*)"

# 2.1.1 A category name may contain any of the characters [A-Za-z0-9+_.-].
# It must not begin with a hyphen or a dot.
_cat = r"[\w+][\w+.-]*"

# 2.1.2 A package name may contain any of the characters [A-Za-z0-9+_-].
# It must not begin with a hyphen,
# and must not end in a hyphen followed by one or more digits.
_pkg = r"[\w+][\w+-]*?"

_v = r"(\d+)((\.\d+)*)([a-z]?)((_(pre|p|beta|alpha|rc)\d*)*)"
_rev = r"\d+"
_vr = _v + "(-r(" + _rev + "))?"

_cp = "(" + _cat + "/" + _pkg + "(-" + _vr + ")?)"
_cpv = "(" + _cp + "-" + _vr + ")"
_pv = (
    "(?P<pn>"
    + _pkg
    + "(?P<pn_inval>-"
    + _vr
    + ")?)"
    + "-(?P<ver>"
    + _v
    + ")(-r(?P<rev>"
    + _rev
    + "))?"
)

ver_regexp = re.compile("^" + _vr + "$")
suffix_regexp = re.compile("^(alpha|beta|rc|pre|p)(\\d*)$")
suffix_value = {"pre": -2, "p": 0, "alpha": -4, "beta": -3, "rc": -1}
endversion_keys = ["pre", "p", "alpha", "beta", "rc"]

_slot_re_cache = {}


def _get_slot_re(eapi_attrs: _eapi_attrs) -> typing.Pattern:
    cache_key = eapi_attrs.slot_operator
    slot_re = _slot_re_cache.get(cache_key)
    if slot_re is not None:
        return slot_re

    if eapi_attrs.slot_operator:
        slot_re = _slot + r"(/" + _slot + r")?"
    else:
        slot_re = _slot

    slot_re = re.compile("^" + slot_re + "$", re.VERBOSE | re.UNICODE)

    _slot_re_cache[cache_key] = slot_re
    return slot_re


_pv_re = None


def _get_pv_re(eapi_attrs: _eapi_attrs) -> typing.Pattern:
    global _pv_re
    if _pv_re is not None:
        return _pv_re

    _pv_re = re.compile(r"^" + _pv + r"$", re.VERBOSE | re.UNICODE)

    return _pv_re


def ververify(myver: str, silent: int = 1) -> bool:
    if ver_regexp.match(myver):
        return True
    if not silent:
        print(_("!!! syntax error in version: %s") % myver)
    return False


@lru_cache(1024)
def vercmp(ver1: str, ver2: str, silent: int = 1) -> Optional[int]:
    """
    Compare two versions
    Example usage:
            >>> from portage.versions import vercmp
            >>> vercmp('1.0-r1','1.2-r3')
            negative number
            >>> vercmp('1.3','1.2-r3')
            positive number
            >>> vercmp('1.0_p3','1.0_p3')
            0

    @param pkg1: version to compare with (see ver_regexp in portage.versions.py)
    @type pkg1: string (example: "2.1.2-r3")
    @param pkg2: version to compare againts (see ver_regexp in portage.versions.py)
    @type pkg2: string (example: "2.1.2_rc5")
    @rtype: None or float
    @return:
    1. positive if ver1 is greater than ver2
    2. negative if ver1 is less than ver2
    3. 0 if ver1 equals ver2
    4. None if ver1 or ver2 are invalid (see ver_regexp in portage.versions.py)
    """

    if ver1 == ver2:
        return 0

    match1 = ver_regexp.match(ver1)
    match2 = ver_regexp.match(ver2)

    # checking that the versions are valid
    if not match1 or not match1.groups():
        if not silent:
            print(_("!!! syntax error in version: %s") % ver1)
        return None
    if not match2 or not match2.groups():
        if not silent:
            print(_("!!! syntax error in version: %s") % ver2)
        return None

    # building lists of the version parts before the suffix
    # first part is simple
    list1 = [int(match1.group(1))]
    list2 = [int(match2.group(1))]

    # this part would greatly benefit from a fixed-length version pattern
    if match1.group(2) or match2.group(2):
        vlist1 = match1.group(2)[1:].split(".")
        vlist2 = match2.group(2)[1:].split(".")

        for i in range(0, max(len(vlist1), len(vlist2))):
            # Implcit .0 is given a value of -1, so that 1.0.0 > 1.0, since it
            # would be ambiguous if two versions that aren't literally equal
            # are given the same value (in sorting, for example).
            if len(vlist1) <= i or len(vlist1[i]) == 0:
                list1.append(-1)
                list2.append(int(vlist2[i]))
            elif len(vlist2) <= i or len(vlist2[i]) == 0:
                list1.append(int(vlist1[i]))
                list2.append(-1)
            # Let's make life easy and use integers unless we're forced to use floats
            elif vlist1[i][0] != "0" and vlist2[i][0] != "0":
                list1.append(int(vlist1[i]))
                list2.append(int(vlist2[i]))
            # now we have to use floats so 1.02 compares correctly against 1.1
            else:
                # list1.append(float("0."+vlist1[i]))
                # list2.append(float("0."+vlist2[i]))
                # Since python floats have limited range, we multiply both
                # floating point representations by a constant so that they are
                # transformed into whole numbers. This allows the practically
                # infinite range of a python int to be exploited. The
                # multiplication is done by padding both literal strings with
                # zeros as necessary to ensure equal length.
                max_len = max(len(vlist1[i]), len(vlist2[i]))
                list1.append(int(vlist1[i].ljust(max_len, "0")))
                list2.append(int(vlist2[i].ljust(max_len, "0")))

    # and now the final letter
    # NOTE: Behavior changed in r2309 (between portage-2.0.x and portage-2.1).
    # The new behavior is 12.2.5 > 12.2b which, depending on how you look at,
    # may seem counter-intuitive. However, if you really think about it, it
    # seems like it's probably safe to assume that this is the behavior that
    # is intended by anyone who would use versions such as these.
    if len(match1.group(4)):
        list1.append(ord(match1.group(4)))
    if len(match2.group(4)):
        list2.append(ord(match2.group(4)))

    for i in range(0, max(len(list1), len(list2))):
        if len(list1) <= i:
            return -1
        if len(list2) <= i:
            return 1
        if list1[i] != list2[i]:
            a = list1[i]
            b = list2[i]
            rval = (a > b) - (a < b)
            return rval

    # main version is equal, so now compare the _suffix part
    list1 = match1.group(5).split("_")[1:]
    list2 = match2.group(5).split("_")[1:]

    for i in range(0, max(len(list1), len(list2))):
        # Implicit _p0 is given a value of -1, so that 1 < 1_p0
        if len(list1) <= i:
            s1 = ("p", "-1")
        else:
            s1 = suffix_regexp.match(list1[i]).groups()
        if len(list2) <= i:
            s2 = ("p", "-1")
        else:
            s2 = suffix_regexp.match(list2[i]).groups()
        if s1[0] != s2[0]:
            a = suffix_value[s1[0]]
            b = suffix_value[s2[0]]
            rval = (a > b) - (a < b)
            return rval
        if s1[1] != s2[1]:
            # it's possible that the s(1|2)[1] == ''
            # in such a case, fudge it.
            try:
                r1 = int(s1[1])
            except ValueError:
                r1 = 0
            try:
                r2 = int(s2[1])
            except ValueError:
                r2 = 0
            rval = (r1 > r2) - (r1 < r2)
            if rval:
                return rval

    # the suffix part is equal to, so finally check the revision
    if match1.group(9):
        r1 = int(match1.group(9))
    else:
        r1 = 0
    if match2.group(9):
        r2 = int(match2.group(9))
    else:
        r2 = 0
    rval = (r1 > r2) - (r1 < r2)
    return rval


def pkgcmp(pkg1: tuple[str, str, str], pkg2: tuple[str, str, str]) -> Optional[int]:
    """
    Compare 2 package versions created in pkgsplit format.

    Example usage:
            >>> from portage.versions import *
            >>> pkgcmp(pkgsplit('test-1.0-r1'),pkgsplit('test-1.2-r3'))
            -1
            >>> pkgcmp(pkgsplit('test-1.3'),pkgsplit('test-1.2-r3'))
            1

    @param pkg1: package to compare with
    @type pkg1: list (example: ['test', '1.0', 'r1'])
    @param pkg2: package to compare againts
    @type pkg2: list (example: ['test', '1.0', 'r1'])
    @rtype: None or integer
    @return:
            1. None if package names are not the same
            2. 1 if pkg1 is greater than pkg2
            3. -1 if pkg1 is less than pkg2
            4. 0 if pkg1 equals pkg2
    """
    if pkg1[0] != pkg2[0]:
        return None
    return vercmp("-".join(pkg1[1:]), "-".join(pkg2[1:]))


def _pkgsplit(mypkg: str, eapi: Any = None) -> Optional[tuple[str, str, str]]:
    """
    @param mypkg: pv
    @return:
    1. None if input is invalid.
    2. (pn, ver, rev) if input is pv
    """
    m = _get_pv_re(_get_eapi_attrs(eapi)).match(mypkg)
    if m is None:
        return None

    if m.group("pn_inval") is not None:
        # package name appears to have a version-like suffix
        return None

    rev = m.group("rev")
    if rev is None:
        rev = "0"
    rev = "r" + rev

    return (m.group("pn"), m.group("ver"), rev)


_cat_re = re.compile(f"^{_cat}$", re.UNICODE)
_missing_cat = "null"


@lru_cache(10240)
def catpkgsplit(
    mydata: Union[str, "_pkg_str"],
    silent: int = 1,
    eapi: Any = None,
) -> Optional[tuple[str, ...]]:
    """
    Takes a Category/Package-Version-Rev and returns a list of each.

    @param mydata: Data to split
    @type mydata: string
    @param silent: suppress error messages
    @type silent: Boolean (integer)
    @rype: list
    @return:
    1.  If each exists, it returns [cat, pkgname, version, rev]
    2.  If cat is not specificed in mydata, cat will be "null"
    3.  if rev does not exist it will be '-r0'
    """
    try:
        return mydata.cpv_split
    except AttributeError:
        pass
    mysplit = mydata.split("/", 1)
    p_split = None
    if len(mysplit) == 1:
        cat = _missing_cat
        p_split = _pkgsplit(mydata, eapi=eapi)
    elif len(mysplit) == 2:
        cat = mysplit[0]
        if _cat_re.match(cat) is not None:
            p_split = _pkgsplit(mysplit[1], eapi=eapi)
    if not p_split:
        return None
    retval = (cat, p_split[0], p_split[1], p_split[2])
    return retval


class _pkg_str(str):
    """
    This class represents a cpv. It inherits from str and has attributes
    that cache results for use by functions like catpkgsplit and cpv_getkey
    which are called frequently (especially in match_from_list).  Instances
    are typically created in dbapi.cp_list() or the Atom contructor, and
    propagate from there. Generally, code that pickles these objects will
    manually convert them to a plain unicode object first.

    Instances of this class will have missing attributes for metadata that
    has not been passed into the constructor. The missing attributes are
    used to distinguish missing metadata values from undefined metadata values.
    For example, the repo attribute will be missing if the 'repository' key
    is missing from the metadata dictionary.
    """

    def __new__(
        cls,
        cpv: str,
        metadata: Optional[dict[str, Any]] = None,
        settings: Any = None,
        eapi: Any = None,
        repo: Optional[str] = None,
        slot: Optional[str] = None,
        build_time: Optional[int] = None,
        build_id: Optional[str] = None,
        file_size: Optional[int] = None,
        mtime: Optional[int] = None,
        db: Any = None,
    ):
        return str.__new__(cls, cpv)

    def __init__(
        self,
        cpv: str,
        metadata: Optional[dict[str, Any]] = None,
        settings: Any = None,
        eapi: Any = None,
        repo: Optional[str] = None,
        slot: Optional[str] = None,
        build_time: Optional[int] = None,
        build_id: Optional[str] = None,
        file_size: Optional[int] = None,
        mtime: Optional[int] = None,
        db: Any = None,
    ):
        if not isinstance(cpv, str):
            # Avoid TypeError from str.__init__ with PyPy.
            cpv = _unicode_decode(cpv)
        str.__init__(cpv)
        if metadata is not None:
            self.__dict__["_metadata"] = metadata
            slot = metadata.get("SLOT", slot)
            repo = metadata.get("repository", repo)
            eapi = metadata.get("EAPI", eapi)
            build_time = metadata.get("BUILD_TIME", build_time)
            file_size = metadata.get("SIZE", file_size)
            build_id = metadata.get("BUILD_ID", build_id)
            mtime = metadata.get("_mtime_", mtime)
        if settings is not None:
            self.__dict__["_settings"] = settings
        if db is not None:
            self.__dict__["_db"] = db
        if eapi is not None:
            self.__dict__["eapi"] = eapi

        self.__dict__["build_time"] = self._long(build_time, 0)
        self.__dict__["file_size"] = self._long(file_size, None)
        self.__dict__["build_id"] = self._long(build_id, None)
        self.__dict__["mtime"] = self._long(mtime, None)
        self.__dict__["cpv_split"] = catpkgsplit(cpv, eapi=eapi)
        if self.cpv_split is None:
            raise InvalidData(cpv)
        self.__dict__["cp"] = self.cpv_split[0] + "/" + self.cpv_split[1]
        if self.cpv_split[-1] == "r0" and cpv[-3:] != "-r0":
            self.__dict__["version"] = "-".join(self.cpv_split[2:-1])
        else:
            self.__dict__["version"] = "-".join(self.cpv_split[2:])
        # for match_from_list introspection
        self.__dict__["cpv"] = self
        if slot is not None:
            eapi_attrs = _get_eapi_attrs(eapi)
            slot_match = _get_slot_re(eapi_attrs).match(slot)
            if slot_match is None:
                # Avoid an InvalidAtom exception when creating SLOT atoms
                self.__dict__["slot"] = "0"
                self.__dict__["sub_slot"] = "0"
                self.__dict__["slot_invalid"] = slot
            else:
                if eapi_attrs.slot_operator:
                    slot_split = slot.split("/")
                    self.__dict__["slot"] = slot_split[0]
                    if len(slot_split) > 1:
                        self.__dict__["sub_slot"] = slot_split[1]
                    else:
                        self.__dict__["sub_slot"] = slot_split[0]
                else:
                    self.__dict__["slot"] = slot
                    self.__dict__["sub_slot"] = slot

        if repo is not None:
            repo = _gen_valid_repo(repo)
            if not repo:
                repo = _unknown_repo
            self.__dict__["repo"] = repo

    def __setattr__(self, name: str, value: Any) -> None:
        raise AttributeError(
            "_pkg_str instances are immutable", self.__class__, name, value
        )

    @staticmethod
    def _long(var: Any, default: int) -> int:
        if var is not None:
            try:
                var = int(var)
            except ValueError:
                if var:
                    var = -1
                else:
                    var = default
        return var

    @property
    def stable(self) -> bool:
        try:
            return self._stable
        except AttributeError:
            try:
                settings = self._settings
            except AttributeError:
                raise AttributeError("stable")
            if not settings.local_config:
                # Since repoman uses different config instances for
                # different profiles, our local instance does not
                # refer to the correct profile.
                raise AssertionError("invalid context")
            stable = settings._isStable(self)
            self.__dict__["_stable"] = stable
            return stable


def pkgsplit(
    mypkg: str, silent: int = 1, eapi: Any = None
) -> Optional[tuple[str, str, str]]:
    """
    @param mypkg: either a pv or cpv
    @return:
    1. None if input is invalid.
    2. (pn, ver, rev) if input is pv
    3. (cp, ver, rev) if input is a cpv
    """
    catpsplit = catpkgsplit(mypkg, eapi=eapi)
    if catpsplit is None:
        return None
    cat, pn, ver, rev = catpsplit
    if cat is _missing_cat and "/" not in mypkg:
        return (pn, ver, rev)
    return (cat + "/" + pn, ver, rev)


def cpv_getkey(mycpv: Union[_pkg_str, str], eapi: Any = None) -> Optional[str]:
    """Calls catpkgsplit on a cpv and returns only the cp."""
    try:
        return mycpv.cp
    except AttributeError:
        pass
    mysplit = catpkgsplit(mycpv, eapi=eapi)
    if mysplit is not None:
        return mysplit[0] + "/" + mysplit[1]

    warnings.warn(
        "portage.versions.cpv_getkey() " + f"called with invalid cpv: '{mycpv}'",
        DeprecationWarning,
        stacklevel=2,
    )

    myslash = mycpv.split("/", 1)
    mysplit = _pkgsplit(myslash[-1], eapi=eapi)
    if mysplit is None:
        return None
    mylen = len(myslash)
    if mylen == 2:
        return myslash[0] + "/" + mysplit[0]
    return mysplit[0]


def cpv_getversion(mycpv: Union[str, _pkg_str], eapi: Any = None) -> Optional[str]:
    """Returns the v (including revision) from an cpv."""
    try:
        return mycpv.version
    except AttributeError:
        pass
    cp = cpv_getkey(mycpv, eapi=eapi)
    if cp is None:
        return None
    return mycpv[len(cp + "-") :]


def cpv_sort_key(eapi: Any = None) -> Any:
    """
    Create an object for sorting cpvs, to be used as the 'key' parameter
    in places like list.sort() or sorted(). This calls catpkgsplit() once for
    each cpv and caches the result. If a given cpv is invalid or two cpvs
    have different category/package names, then plain string (> and <)
    comparison is used.

    @rtype: key object for sorting
    @return: object for use as the 'key' parameter in places like
            list.sort() or sorted()
    """

    split_cache = {}

    def cmp_cpv(cpv1: Any, cpv2: Any) -> int:
        split1 = split_cache.get(cpv1, False)
        if split1 is False:
            split1 = None
            try:
                split1 = cpv1.cpv
            except AttributeError:
                try:
                    split1 = _pkg_str(cpv1, eapi=eapi)
                except InvalidData:
                    pass
            split_cache[cpv1] = split1

        split2 = split_cache.get(cpv2, False)
        if split2 is False:
            split2 = None
            try:
                split2 = cpv2.cpv
            except AttributeError:
                try:
                    split2 = _pkg_str(cpv2, eapi=eapi)
                except InvalidData:
                    pass
            split_cache[cpv2] = split2

        if split1 is None or split2 is None or split1.cp != split2.cp:
            return (cpv1 > cpv2) - (cpv1 < cpv2)

        return vercmp(split1.version, split2.version)

    return cmp_sort_key(cmp_cpv)


def catsplit(mydep: str) -> list[str]:
    return mydep.split("/", 1)


def best(mymatches: Sequence[Any], eapi: Any = None) -> Any:
    """Accepts None arguments; assumes matches are valid."""
    if not mymatches:
        return ""
    if len(mymatches) == 1:
        return mymatches[0]
    bestmatch = mymatches[0]
    try:
        v2 = bestmatch.version
    except AttributeError:
        v2 = _pkg_str(bestmatch, eapi=eapi).version
    for x in mymatches[1:]:
        try:
            v1 = x.version
        except AttributeError:
            v1 = _pkg_str(x, eapi=eapi).version
        if vercmp(v1, v2) > 0:
            bestmatch = x
            v2 = v1
    return bestmatch
